package com.congwulin.wechat.entity.common;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 一主多子抽象实体类
 */
public abstract class AbstractMeBillVO {
    /**
     * 区域实体类索引
     */
    private Map<Class<?>, String> areaClassIndex = new HashMap<Class<?>, String>();

    /**
     * 区域名称索引
     */
    private Map<String, Class<?>> areaNameIndex = new HashMap<String, Class<?>>();

    /**
     * 子实体索引
     */
    private Map<Class<?>, List<SuperVO>> childList = new HashMap<Class<?>, List<SuperVO>>();

    /**
     * 获取子实体的外键(主实体主键)
     */
    private Map<Class<?>, String> fkIndex = new HashMap<Class<?>, String>();

    /**
     * 主实体
     */
    private SuperVO head;

    /**
     * 一主多子抽象实体构造函数
     */
    protected AbstractMeBillVO() {
        this.setVOClazz();
        this.initChildrenFK();
    }

    /**
     * 获取区域名称索引
     *
     * @return 区域名称索引
     */
    public Map<String, Class<?>> getAreaNameIndex() {
        return this.areaNameIndex;
    }

    /**
     * 根据实体类获取子实体列表
     *
     * @param clazz 实体类
     * @return 子实体列表
     */
    public List<SuperVO> getChildren(Class<?> clazz) {
        return this.childList.get(clazz);
    }


    /**
     * 获取所有子表vo数据
     * @return
     */
    public Map<Class<?>, List<SuperVO>> getChildrens(){
        return this.childList;
    }

    /**
     * 获取子实体的外键(主实体主键)
     *
     * @return 子实体类型和外键的映射
     */
    public Map<Class<?>, String> getChildrenFK() {
        return this.fkIndex;
    }

    /**
     * 获取主实体
     *
     * @return 主实体
     */
    public SuperVO getHead() {
        return this.head;
    }

    /**
     * 设置子实体列表
     *
     * @param clazz 实体类
     * @param list 实体列表
     */
    public void setChildren(Class<?> clazz, List<?> list) {
        List<SuperVO> volist = (List<SuperVO>) list;
        this.childList.put(clazz, volist);
    }

    /**
     * 设置主实体
     *
     * @param head 主实体
     */
    public void setHead(SuperVO head) {
        this.head = head;
    }

    /**
     * 添加区域名称和实体类的映射
     *
     * @param name 区域名称
     * @param clazz 实体类
     */
    protected void addArea(String name, Class<?> clazz) {
        this.areaClassIndex.put(clazz, name);
        this.areaNameIndex.put(name, clazz);
    }

    /**
     * 设置子实体的外键
     *
     * @param clazz 子实体类型
     * @param fk 外键属性名
     */
    protected void addFK(Class<?> clazz, String fk) {
        this.fkIndex.put(clazz, fk);
    }

    /**
     * 初始化子实体的外键
     */
    protected abstract void initChildrenFK();

    /**
     * 设置子实体类
     */
    protected abstract void setVOClazz();
}
